tensor([[[[[0.5200, 0.5100],
           [0.0000, 0.0000]],

          [[0.3100, 0.0000],
           [0.4600, 0.2100]]],


         [[[0.4800, 0.4800],
           [0.4800, 0.0000]],

          [[0.2800, 0.4600],
           [0.4900, 0.5100]]]],



        [[[[0.4800, 0.4900],
           [0.4500, 0.0000]],

          [[0.3400, 0.5000],
           [0.0000, 0.2400]]],


         [[[0.0000, 0.5200],
           [0.0000, 0.0000]],

          [[0.1700, 0.1100],
           [0.2200, 0.1500]]]],



        [[[[0.0000, 0.0000],
           [0.5500, 0.5200]],

          [[0.0000, 0.0000],
           [0.5400, 0.2800]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.1300, 0.1500],
           [0.1700, 0.2000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.4800]],

          [[0.3500, 0.5000],
           [0.0000, 0.2700]]],


         [[[0.5200, 0.0000],
           [0.5200, 1.0000]],

          [[0.4200, 0.2800],
           [0.1200, 0.1400]]]]], grad_fn=<RoundBackward1>)